package com.note.feng.leetcode.algorithms.easy.ten;

import java.util.ArrayList;
import java.util.List;

public class ThousandSeventyEight {
    /**
     * 1078 Bigram 分词
     * 给出第一个词 first 和第二个词 second，考虑在某些文本 text 中可能以 "first second third" 形式出现的情况，其中 second 紧随 first 出现，third 紧随 second 出现。
     *
     * 对于每种这样的情况，将第三个词 "third" 添加到答案中，并返回答案。
     *
     * 示例 1：
     *
     * 输入：text = "alice is a good girl she is a good student", first = "a", second = "good"
     * 输出：["girl","student"]
     * 示例 2：
     *
     * 输入：text = "we will we will rock you", first = "we", second = "will"
     * 输出：["we","rock"]
     *
     * 提示：
     *
     * 1 <= text.length <= 1000
     * text 由小写英文字母和空格组成
     * text 中的所有单词之间都由 单个空格字符 分隔
     * 1 <= first.length, second.length <= 10
     * first 和 second 由小写英文字母组成
     *
     * 来源：力扣（LeetCode）
     * 链接：https://leetcode.cn/problems/occurrences-after-bigram
     */
    /**
     * 解法：遍历
     * 将 text 以空格分割成数组，再遍历，若连续两个单词同时匹配 first 和 second 则后一个单词就是需要的值
     * @param text
     * @param first
     * @param second
     * @return
     */
    public String[] findOcurrences(String text, String first, String second) {
        List<String> list = new ArrayList<>();
        String [] texts = text.split(" ");
        for (int i = 0; i < texts.length - 2; i ++){
            if(first.equals(texts[i]) && second.equals(texts[i + 1])){
                list.add(texts[i + 2]);
            }
        }
        if(list.isEmpty()){
            return new String[0];
        }
        String [] res = new String[list.size()];
        for (int i = 0; i < list.size(); i ++){
            res[i] = list.get(i);
        }
        return res;
    }
}
